-- 2022-8-5

---[[出现频率最高的单词

    -- 本章中 开发一个读取并输出一段文本中出现频率最高的单词程序
    -- 使用了诸如迭代器和匿名函数这样的高级特性

    -- 该程序的主要数据结构是一个记录文本中出现的每一个单词及其出现次数之间的关系的表
    -- 使用这个数据结构 该程序可以完成3个主要任务
    --      读取文本并计算每一个单词的出现次数
    --      按照出现次数的降序对单词列表进行排序
    --      输出有序列表的前n个元素

    -- 要读取文本，可以遍历每一行，然后遍历每一行的每一个单词
    -- 对于我们读取的每一个单词，增加对于计数器的值
    local counter = {}
    for line in io.lines() do
        for word in string.gmatch(line,"%w+") do
            counter[word] = (counter[word] or 0) + 1
        end
    end
    -- 这里，我们使用模式'%w+'来描述单词，也就是一个或多个字母或数字

    -- 下一步就是对单词列表进行排序
    -- 不过 Lua中并没有可以用来排序的单词列表 尽管如此 使用表counter中作为键的单词来创建一个列表还是很简单的
    local words = {}
    for w in ipairs(counter) do
        words[#words+1] = w
    end

    -- 一旦有了单词列表就可以使用函数table.sort对其进行排序
    table.sort(words,function (w1, w2)
        return counter[w1] > counter[w2] or counter[w1] == counter[w2]
    end)
    -- 记住，排序函数必须在w1位于w2之前时返回真，计数值越大的单词排的越前，具有相同计数值的单词则按照字母顺序排序

    -- 示例 11.1 展示了完整的代码
    
    -- 示例 11.1 统计单词出现频率的程序
    local counter = {}
    for line in io.lines() do
        for word in string.gmatch(line,"%w+") do
            counter[word] = (counter[word] or 0) + 1
        end
    end
    local words = {} -- 文本中所有单词的列表

    for w in ipairs(counter) do
        words[#words+1] = w
    end

    table.sort(words,function (w1, w2)
        return counter[w1] > counter[w2] or counter[w1] == counter[w2] and w1 < w2
    end)

    -- 要输出的字数
    local n = math.min(tonumber(arg[1] or math.huge,#words))
    for i = 1, n, 1 do
        io.write(words[i],"\t",counter[words[i]],"\n")
    end
    -- 输出结果
--]]  